const KIM_CHAT_API_URL = 'https://api.moonshot.cn/v1/chat/completions';

export const chat = async (
    messages, 
    api_url=KIM_CHAT_API_URL, 
    api_key=import.meta.env.VITE_KIMI_API_KEY,
    model='moonshot-v1-auto'
) => {
    try {
        const response = await fetch(api_url, {
            method: 'POST',
            headers: {
                'Content-Type': 'application/json',
                'Authorization': `Bearer ${api_key}`
            },
            body: JSON.stringify({
                model,
                messages,
                stream: false,
            })
        })
        const data = await response.json();
        return {
            code: 0,
            data: {
                role: 'assistant',
                content: data.choices[0].message.content
            }
            
        }
    } catch(err) {
        return {
            code: 0,
            msg: '出错了...'
        }
   } 
}

export const kimiChat = async (messages) => {
    const res = await chat(
        messages,
        KIM_CHAT_API_URL,
        import.meta.env.VITE_KIMI_API_KEY,
        'moonshot-v1-auto'
    )
    return res;
}

export const priceComparisonChat = async (messages) => {
    // 添加系统提示，强化比价角色
    const systemPrompt = {
        role: 'system',
        content: '你是一个专业的AI比价助手，负责为用户提供准确、及时的商品价格比较信息。\n当用户询问价格时，你需要:\n1. 尽可能提供多个平台的价格对比\n2. 指出最低价格的平台\n3. 如果有优惠活动，也请一并告知\n4. 保持信息客观准确'
    };
    
    // 合并系统提示和用户消息
    const enhancedMessages = [systemPrompt, ...messages];
    
    // 调用聊天API
    const res = await chat(
        enhancedMessages,
        KIM_CHAT_API_URL,
        import.meta.env.VITE_KIMI_API_KEY,
        'moonshot-v1-auto'
    );
    
    return res;
}

export const generateAvatar = async (text) => {
    // 设计prompt 
    const prompt = `
    你是一位漫画设计师，需要为用户设计头像，主打日漫风格。
    用户的信息是${text}
    要求有个性，有设计感。
    `
}